working with functions python class 12